package com.zy.explore.wechat.service;

import com.zy.explore.wechat.assembler.JpaExampleAssembler;
import com.zy.explore.wechat.frm.CommonIntIdFrm;
import com.zy.explore.wechat.frm.ExampleFrm;
import com.zy.explore.wechat.vo.ExampleVO;
import com.zy.explore.wechat.vo.PageVO;
import com.zy.explore.wechat.domain.jpa.entity.JpaExampleEntity;

import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageImpl;
import org.springframework.data.domain.Sort;
import org.springframework.data.jpa.domain.Specification;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

import java.util.ArrayList;
import java.util.List;

/**
 * description Jpa示例服务实现类
 *
 * @author ZY
 * @date 2021年05月15日 22:19
 */
@Service
public class JpaExampleService {

    /**
     * 新增数据
     *
     * @param frm 待新增的frm对象
     * @return 成功返回新增后的VO对象，失败返回null
     */
    public ExampleVO insert(ExampleFrm frm) {
        JpaExampleEntity entity = JpaExampleAssembler.convertFromFrm(frm);
        entity.save();
        return JpaExampleAssembler.convertToVO(entity);
    }

    /**
     * 删除数据
     *
     * @param frm 主键frm对象
     */
    public Integer delete(CommonIntIdFrm frm) {
        JpaExampleEntity entity = JpaExampleEntity.builder().id(frm.getId()).build();
        entity.remove();
        return 1;
    }

    /**
     * 更新数据
     *
     * @param frm 待更新的frm对象
     * @return 成功返回更新后的VO对象，失败返回null
     */
    public ExampleVO update(ExampleFrm frm) {
        JpaExampleEntity entity = JpaExampleAssembler.convertFromFrm(frm);
        entity.updateSelective();
        return JpaExampleAssembler.convertToVO(entity);
    }

    /**
     * 通过主键查询单条数据
     *
     * @param frm 主键Frm对象
     * @return 成功返回对应ID的VO对象，失败返回null
     */
    public ExampleVO getById(CommonIntIdFrm frm) {
        JpaExampleEntity entity = JpaExampleEntity.builder().id(frm.getId()).build();
        return JpaExampleAssembler.convertToVO(entity.findById());
    }

    /**
     * 根据条件获取数据集
     *
     * @param frm 查询条件对象
     * @return 成功返回满足条件的数据集合，失败返回空集合
     */
    public List<ExampleVO> getList(ExampleFrm frm) {
        List<ExampleVO> retList = new ArrayList<>();
        // 转换查询条件
        Specification<JpaExampleEntity> specification = JpaExampleAssembler.convertToSpecification(frm);
        // 转换排序条件
        Sort sort = JpaExampleAssembler.getSort(frm);
        JpaExampleEntity entity = JpaExampleEntity.builder().build();
        List<JpaExampleEntity> list = entity.findAll(specification, sort);
        if (!CollectionUtils.isEmpty(list)) {
            list.forEach(x -> retList.add(JpaExampleAssembler.convertToVO(x)));
        }
        return retList;

    }

    /**
     * 根据条件查询数据分页
     *
     * @param frm 查询条件参数
     * @return 成功返回满足条件的分页，失败返回空页
     */
    public PageVO<ExampleVO> getPageInfo(ExampleFrm frm) {
        // 转换查询条件
        Specification<JpaExampleEntity> specification = JpaExampleAssembler.convertToSpecification(frm);
        // 转换排序条件
        Sort sort = JpaExampleAssembler.getSort(frm);
        JpaExampleEntity entity = JpaExampleEntity.builder().build();
        Page<JpaExampleEntity> page = entity.findAll(specification, frm.getPageNum(), frm.getPageSize(), sort);
        Page<ExampleVO> retPage = new PageImpl<>(new ArrayList<>(), page.getPageable(), page.getTotalElements());
        if (page.hasContent()) {
            List<ExampleVO> voList = new ArrayList<>();
            page.getContent().forEach(x -> voList.add(JpaExampleAssembler.convertToVO(x)));
            retPage = new PageImpl<>(voList, page.getPageable(), page.getTotalElements());
        }
        return new PageVO<>(retPage);
    }

}
